package com.dzu.easy;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;

/**
 * @author ZhaoDong
 * @date 2022/8/3 9:42
 * @description leetcode
 * 219. 存在重复元素 II
 * 给你一个整数数组 nums 和一个整数 k ，判断数组中是否存在两个 不同的索引 i 和 j ，满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在，返回 true ；否则，返回 false 。
 *
 *  
 *
 * 示例 1：
 *
 * 输入：nums = [1,2,3,1], k = 3
 * 输出：true
 * 示例 2：
 *
 * 输入：nums = [1,0,1,1], k = 1
 * 输出：true
 * 示例 3：
 *
 * 输入：nums = [1,2,3,1,2,3], k = 2
 * 输出：false
 *  
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/contains-duplicate-ii
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 */
public class Test219 {

    public static void main(String[] args) {

        boolean b = containsNearbyDuplicate(new int[]{1, 2, 3, 1}, 3);
        System.out.println(b);
    }


    /**
     * 执行用时：
     * 18 ms
     * , 在所有 Java 提交中击败了
     * 53.90%
     * 的用户
     * 内存消耗：
     * 53.4 MB
     * , 在所有 Java 提交中击败了
     * 8.12%
     * 的用户
     * @param nums
     * @param k
     * @return
     */
    public static boolean containsNearbyDuplicate2(int[] nums, int k) {
        HashSet<Integer> set = new HashSet<>();
        for(int i = 0; i < nums.length; i++) {
            if(set.contains(nums[i])) {
                return true;
            }
            set.add(nums[i]);
            if(set.size() > k) {
                set.remove(nums[i - k]);
            }
        }
        return false;
    }

        /**
         * 执行用时：35 ms在所有 Java 提交中击败了14.79%的用户
         * 内存消耗：
         * 52.3 MB
         * , 在所有 Java 提交中击败了
         * 29.45%
         * 的用户
         * @param nums
         * @param k
         * @return
         */
    public static boolean containsNearbyDuplicate(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }

        for (int i = 0; i < nums.length; i++) {
            int num = nums[i];
            Integer length = map.get(num);
            if (length >= 2) {

                for (int j = i + 1; j < nums.length; j++) {
                    int num2 = nums[j];
                    if (num == num2 && Math.abs(i - j) <= k) {
                        return true;
                    }
                }
            }
        }
        return false;

    }
}
